#include "ota_http.h"
#include <curl/curl.h>
#include "log/log.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

/// @brief 接收响应数据的回调函数
/// @param ptr 指向响应数据的指针
/// @param size     每个数据块的大小
/// @param nmemb     数据块的数量
/// @param userdata     用户自定义数据
size_t write_callback(void *ptr, size_t size, size_t nmemb, void *userdata)
{
    char *json_buf = (char *)userdata;
    // 数据长度
    int data_len = size * nmemb;
    // 将数据写入容器
    memcpy(json_buf, ptr, data_len);
    // 写结束符号
    json_buf[data_len] = '\0';
    return data_len;
}
char *ota_http_getJSON(char *url)
{
    // 初始化curl
    CURL *curl = curl_easy_init();
    // 配置url
    curl_easy_setopt(curl, CURLOPT_URL, url); // option
    // 指定接收响应数据的回调函数
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_callback); // option
    // 指定接收响应数据的容器，交给回调函数。一旦收到数据，将数据写入容器
    char *json_buf = malloc(1024); // 假定json数据不超过1024字节
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, json_buf);
    // 执行请求，如果失败，释放资源，返回null
    CURLcode res = curl_easy_perform(curl);
    if (res != CURLE_OK)
    {
        free(json_buf);
        curl_easy_cleanup(curl);
        log_error("curl_easy_perform() failed: %s\n", curl_easy_strerror(res));
        return NULL;
    }
    // 请求成功，释放资源
    curl_easy_cleanup(curl);
    // 返回容器数据
    log_debug("json_buf: %s\n", json_buf);
    return json_buf;
}

int ota_http_download(char *url, char *file_name)
{
    // 初始化curl
    CURL *curl = curl_easy_init();
    // 指定url
    curl_easy_setopt(curl, CURLOPT_URL, url);
    // 指定得到返回文件的回调函数
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, fwrite);
    // 指定得到返回文件的回调函数的参数，即文件指针
    FILE *fp = fopen(file_name, "wb"); // write binary
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp);
    // 执行请求，如果失败，释放资源，返回-1
    CURLcode res = curl_easy_perform(curl);
    if (res != CURLE_OK)
    {
        fclose(fp);
        curl_easy_cleanup(curl);
        log_error("curl_easy_perform() failed: %s\n", curl_easy_strerror(res));
        return -1;
    }
    // 请求成功，释放资源
    fclose(fp);
    curl_easy_cleanup(curl);
    // 返回容器数据
    return 0;
}
